import plotly
import plotly.express as px
import pandas as pd
import warnings
warnings.filterwarnings('ignore')
df = pd.read_csv("HECoS_CAH_Mappings.csv")
plotly.offline.init_notebook_mode()
fig = px.treemap(df,
path=[px.Constant("All subjects"),
"CAH1_Label",
"CAH2_Label",
"CAH3_Label",
"HECoS_Label"
],
)
fig.update_layout(
overwrite=True,
margin_b=20,
margin_l=20,
margin_r=25,
margin_t=0,
)
fig